hackerrank map and lambda function problem solution in python programming